package com.jcfk.eam.service.mt;

import com.jcfk.common.business.service.BaseService;
import com.jcfk.common.web.domain.DataPage;
import com.jcfk.common.web.domain.RetResult;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.jcfk.eam.dao.mt.EamMtUseApplyDao;
import com.jcfk.eam.domain.dto.base.EamProductAssetModelDTO;
import com.jcfk.eam.domain.dto.mt.EamMtUseApplyDTO;
import com.jcfk.eam.domain.dto.mt.EamMtUseApplyDetailDTO;
import com.jcfk.eam.domain.po.base.EamAssetType;
import com.jcfk.eam.domain.po.mt.EamMtUseApply;
import com.jcfk.eam.domain.po.mt.EamMtUseApplyDetail;
import com.jcfk.eam.domain.query.mt.EamMtUseApplyQuery;
import com.jcfk.eam.service.base.EamAssetTypeService;
import com.jcfk.system.api.utils.CodeBuildUtils;
import org.springframework.stereotype.Service;

import org.springframework.transaction.annotation.Transactional;
import org.springframework.beans.factory.annotation.Autowired;
import com.jcfk.common.core.utils.BeanUtils;
import cn.hutool.core.util.StrUtil;
import com.jcfk.common.security.utils.SecurityUtils;
import com.alibaba.druid.util.StringUtils;

import static com.jcfk.common.core.utils.BeanUtils.convertList;

import java.util.*;

import java.util.stream.Collectors;

/**
 * 量检具领用申请
 *
 * @author libinghai
 * @email 35738025@gqq.com
 * @date 2024-09-08
 */
@Service
public class EamMtUseApplyService extends BaseService<EamMtUseApplyDao, EamMtUseApply> {


    @Autowired
    private EamMtUseApplyDetailService eamMtUseApplyDetailService;


    @Autowired
    private EamAssetTypeService eamAssetTypeService;

    /**
     * EamMtUseApply分页查询
     *
     * @param page  分页设置
     * @param query 查询条件
     * @return 分页查询结果
     */
    public DataPage<EamMtUseApplyDTO> queryByPage(DataPage<EamMtUseApplyDTO> page, EamMtUseApplyQuery query) {
        // 分页查询
        page = this.baseMapper.queryByPage(page, query);
        return page;
    }

    /**
     * 根据条件获取单表信息列表
     *
     * @param query 查询条件
     * @return List<EamMtUseApplyDTO>
     */
    public List<EamMtUseApplyDTO> getList(EamMtUseApplyQuery query) {
        // 查询条件
        QueryWrapper<EamMtUseApply> qw = new QueryWrapper<>();
        qw.lambda().eq(query.getApplyType() != null, EamMtUseApply::getApplyType, query.getApplyType());
        qw.lambda().eq(!StrUtil.isEmpty(query.getReceiveDepartmentId()), EamMtUseApply::getReceiveDepartmentId, query.getReceiveDepartmentId());
        qw.lambda().eq(!StrUtil.isEmpty(query.getReceivePersonId()), EamMtUseApply::getReceivePersonId, query.getReceivePersonId());
        qw.lambda().eq(!StrUtil.isEmpty(query.getCurrentProductId()), EamMtUseApply::getCurrentProductId, query.getCurrentProductId());
        qw.lambda().eq(!StrUtil.isEmpty(query.getChangeProductId()), EamMtUseApply::getChangeProductId, query.getChangeProductId());
        qw.lambda().eq(!StrUtil.isEmpty(query.getUsePlaceId()), EamMtUseApply::getUsePlaceId, query.getUsePlaceId());
        qw.lambda().ge(!StrUtil.isEmpty(query.getApplyDateStart()), EamMtUseApply::getApplyDate, query.getApplyDateStart());
        qw.lambda().le(!StrUtil.isEmpty(query.getApplyDateEnd()), EamMtUseApply::getApplyDate, query.getApplyDateEnd());
        // 查询数据
        List<EamMtUseApply> list = super.list(qw);

        return convertList(list, EamMtUseApplyDTO.class);
    }

    /**
     * 获取根据关联id详情表列表
     *
     * @param query 查询条件
     * @return List<EamMtUseApplyDTO>
     */
    public List<EamMtUseApplyDetailDTO> getDetails(EamMtUseApplyQuery query) {
        // 查询条件
        QueryWrapper<EamMtUseApplyDetail> qw = new QueryWrapper<>();
        qw.lambda().eq(!StrUtil.isEmpty(query.getId()), EamMtUseApplyDetail::getApplyId, query.getId());

        // 查询数据
        List<EamMtUseApplyDetail> list = eamMtUseApplyDetailService.list(qw);

        return convertList(list, EamMtUseApplyDetailDTO.class);
    }


    /**
     * 判断EamMtUseApply是已经存在
     *
     * @param dto
     * @return
     */
    public RetResult<String> isValid(EamMtUseApplyDTO dto) {
        QueryWrapper<EamMtUseApply> queryWrapper = new QueryWrapper<>();
        return RetResult.ok();
    }

    /**
     * 插入量检具领用申请
     *
     * @param dto 插入对象
     * @return
     */
    @Transactional(rollbackFor = Exception.class)
    public void insert(EamMtUseApplyDTO dto) {
        // 保存
        EamMtUseApply item = new EamMtUseApply();
        BeanUtils.copyProperties(dto, item);
        if (item.getApplyType() == 1)
            item.setRecordCode(CodeBuildUtils.buildCode("eam_sq", ""));
        else
            item.setRecordCode(CodeBuildUtils.buildCode("eam_hx", ""));

        item.setReceivePersonId(SecurityUtils.getUserId());
        item.setReceivePersonName(SecurityUtils.getUserName());
        this.setDefaultValue(item);
        super.save(item);

        if (dto.getApplyType() == 1) {
            List<EamMtUseApplyDetail> eamMtUseApplyDetails = new ArrayList<>();
            for (EamProductAssetModelDTO currentDetail : dto.getApplyDetails()) {
                EamMtUseApplyDetail eamMtUseApplyDetail = new EamMtUseApplyDetail();
                eamMtUseApplyDetail.setApplyId(item.getId());
                eamMtUseApplyDetail.setAssetTypeId(currentDetail.getAssetTypeId());
                eamMtUseApplyDetail.setApplyAmount(currentDetail.getApplyAmount());
                eamMtUseApplyDetails.add(eamMtUseApplyDetail);
            }
            if (eamMtUseApplyDetails.size() > 0)
                eamMtUseApplyDetailService.saveBatch(eamMtUseApplyDetails);
        } else {
            List<EamMtUseApplyDetail> eamMtUseApplyDetails = new ArrayList<>();
            for (EamProductAssetModelDTO currentDetail : dto.getCurrentDetails()) {
                EamMtUseApplyDetail eamMtUseApplyDetail = new EamMtUseApplyDetail();
                eamMtUseApplyDetail.setApplyId(item.getId());
                eamMtUseApplyDetail.setAssetTypeId(currentDetail.getAssetTypeId());
                eamMtUseApplyDetail.setApplyAmount(currentDetail.getApplyAmount());
                if (dto.getApplyType() == 2)
                    eamMtUseApplyDetail.setDetailType(1);
                eamMtUseApplyDetails.add(eamMtUseApplyDetail);
            }

            if (eamMtUseApplyDetails.size() > 0)
                eamMtUseApplyDetailService.saveBatch(eamMtUseApplyDetails);

            List<EamMtUseApplyDetail> changeDetails = new ArrayList<>();
            for (EamProductAssetModelDTO currentDetail : dto.getChangeDetails()) {
                EamMtUseApplyDetail eamMtUseApplyDetail = new EamMtUseApplyDetail();
                eamMtUseApplyDetail.setApplyId(item.getId());
                eamMtUseApplyDetail.setAssetTypeId(currentDetail.getAssetTypeId());
                eamMtUseApplyDetail.setApplyAmount(currentDetail.getApplyAmount());
                if (dto.getApplyType() == 2)
                    eamMtUseApplyDetail.setDetailType(2);
                changeDetails.add(eamMtUseApplyDetail);
            }

            if (changeDetails.size() > 0)
                eamMtUseApplyDetailService.saveBatch(changeDetails);

            List<EamMtUseApplyDetail> applyDetails = new ArrayList<>();
            for (EamProductAssetModelDTO currentDetail : dto.getApplyDetails()) {
                EamMtUseApplyDetail eamMtUseApplyDetail = new EamMtUseApplyDetail();
                eamMtUseApplyDetail.setApplyId(item.getId());
                eamMtUseApplyDetail.setAssetTypeId(currentDetail.getAssetTypeId());
                eamMtUseApplyDetail.setApplyAmount(currentDetail.getApplyAmount());
                if (dto.getApplyType() == 2)
                    eamMtUseApplyDetail.setDetailType(3);
                applyDetails.add(eamMtUseApplyDetail);
            }

            if (applyDetails.size() > 0)
                eamMtUseApplyDetailService.saveBatch(applyDetails);
        }
    }


    /**
     * 更新量检具领用申请
     *
     * @param dto 更新对象
     * @return
     */
    @Transactional(rollbackFor = Exception.class)
    public void update(EamMtUseApplyDTO dto) {
        // 保存
        EamMtUseApply item = new EamMtUseApply();
        BeanUtils.copyProperties(dto, item);

        // 默认数据
        this.setDefaultValue(item);

        super.updateById(item);

        QueryWrapper<EamMtUseApplyDetail> queryWrapper = new QueryWrapper<>();
        queryWrapper.lambda().eq(EamMtUseApplyDetail::getApplyId, dto.getId());
        eamMtUseApplyDetailService.remove(queryWrapper);

        if (dto.getApplyType() == 1) {
            List<EamMtUseApplyDetail> eamMtUseApplyDetails = new ArrayList<>();
            for (EamProductAssetModelDTO currentDetail : dto.getApplyDetails()) {
                EamMtUseApplyDetail eamMtUseApplyDetail = new EamMtUseApplyDetail();
                eamMtUseApplyDetail.setApplyId(dto.getId());
                eamMtUseApplyDetail.setAssetTypeId(currentDetail.getAssetTypeId());
                eamMtUseApplyDetail.setApplyAmount(currentDetail.getApplyAmount());
                eamMtUseApplyDetails.add(eamMtUseApplyDetail);
            }
            if (eamMtUseApplyDetails.size() > 0)
                eamMtUseApplyDetailService.saveBatch(eamMtUseApplyDetails);
        } else {
            List<EamMtUseApplyDetail> eamMtUseApplyDetails = new ArrayList<>();
            for (EamProductAssetModelDTO currentDetail : dto.getCurrentDetails()) {
                EamMtUseApplyDetail eamMtUseApplyDetail = new EamMtUseApplyDetail();
                eamMtUseApplyDetail.setApplyId(dto.getId());
                eamMtUseApplyDetail.setAssetTypeId(currentDetail.getAssetTypeId());
                eamMtUseApplyDetail.setApplyAmount(currentDetail.getApplyAmount());
                eamMtUseApplyDetail.setDetailType(1);
                eamMtUseApplyDetails.add(eamMtUseApplyDetail);
            }

            if (eamMtUseApplyDetails.size() > 0)
                eamMtUseApplyDetailService.saveBatch(eamMtUseApplyDetails);

            List<EamMtUseApplyDetail> changeDetails = new ArrayList<>();
            for (EamProductAssetModelDTO currentDetail : dto.getChangeDetails()) {
                EamMtUseApplyDetail eamMtUseApplyDetail = new EamMtUseApplyDetail();
                eamMtUseApplyDetail.setApplyId(dto.getId());
                eamMtUseApplyDetail.setAssetTypeId(currentDetail.getAssetTypeId());
                eamMtUseApplyDetail.setApplyAmount(currentDetail.getApplyAmount());
                eamMtUseApplyDetail.setDetailType(2);
                changeDetails.add(eamMtUseApplyDetail);
            }

            if (changeDetails.size() > 0)
                eamMtUseApplyDetailService.saveBatch(changeDetails);

            List<EamMtUseApplyDetail> applyDetails = new ArrayList<>();
            for (EamProductAssetModelDTO currentDetail : dto.getApplyDetails()) {
                EamMtUseApplyDetail eamMtUseApplyDetail = new EamMtUseApplyDetail();
                eamMtUseApplyDetail.setApplyId(dto.getId());
                eamMtUseApplyDetail.setAssetTypeId(currentDetail.getAssetTypeId());
                eamMtUseApplyDetail.setApplyAmount(currentDetail.getApplyAmount());
                eamMtUseApplyDetail.setDetailType(3);
                applyDetails.add(eamMtUseApplyDetail);
            }

            if (applyDetails.size() > 0)
                eamMtUseApplyDetailService.saveBatch(applyDetails);
        }
    }


    /**
     * 设置添加默认值
     *
     * @param item 添加对象
     */
    public void setDefaultValue(EamMtUseApply item) {
        if (StrUtil.isEmpty(item.getCreatedUserId())) {
            item.setCreatedUserId(SecurityUtils.getUserId());
            item.setCreateTime(new Date());
            item.setVersion(1L);
            item.setCreatedUserName(SecurityUtils.getUserName());
        }
        item.setLastUpdatedUserId(SecurityUtils.getUserId());
        item.setLastUpdateTime(new Date());

        item.setLastUpdatedUserName(SecurityUtils.getUserName());
    }


    /**
     * 根据主键获取量检具领用申请信息
     *
     * @param id 主键
     * @return StandardCodeDTO
     */
    public EamMtUseApplyDTO get(String id) {
        // 查询数据
        EamMtUseApply item = this.baseMapper.selectById(id);
        EamMtUseApplyDTO dto = new EamMtUseApplyDTO();
        try {
            BeanUtils.copyProperties(item, dto);
        } catch (RuntimeException e) {
            throw e;
        }

        QueryWrapper<EamMtUseApplyDetail> queryWrapper = new QueryWrapper<>();
        queryWrapper.lambda().eq(EamMtUseApplyDetail::getApplyId, dto.getId());
        List<EamMtUseApplyDetail> list = eamMtUseApplyDetailService.list(queryWrapper);

        List<String> ids = list.stream().map(EamMtUseApplyDetail::getAssetTypeId).collect(Collectors.toList());

        Map<String, EamAssetType> map = new HashMap<>();
        if (ids.size() > 0) {
            QueryWrapper<EamAssetType> eamAssetTypeQueryWrapper = new QueryWrapper<>();
            eamAssetTypeQueryWrapper.lambda().in(EamAssetType::getId, ids);
            List<EamAssetType> eamAssetTypes = eamAssetTypeService.list(eamAssetTypeQueryWrapper);
            for (EamAssetType eamAssetType : eamAssetTypes) {
                map.put(eamAssetType.getId(), eamAssetType);
            }
        }

        for (EamMtUseApplyDetail eamMtUseApplyDetail : list) {
            EamProductAssetModelDTO eamProductAssetModelDTO = new EamProductAssetModelDTO();
            eamProductAssetModelDTO.setApplyAmount(eamMtUseApplyDetail.getApplyAmount());
            eamProductAssetModelDTO.setAssetTypeId(eamMtUseApplyDetail.getAssetTypeId());
            eamProductAssetModelDTO.setAssetTypeName(map.get(eamMtUseApplyDetail.getAssetTypeId()).getAssetTypeName());
            eamProductAssetModelDTO.setAssetTypeCode(map.get(eamMtUseApplyDetail.getAssetTypeId()).getAssetTypeCode());
            if (dto.getApplyType() == 1) {
                dto.getApplyDetails().add(eamProductAssetModelDTO);
            } else {
                if (eamMtUseApplyDetail.getDetailType() == 1)
                    dto.getCurrentDetails().add(eamProductAssetModelDTO);
                if (eamMtUseApplyDetail.getDetailType() == 2)
                    dto.getChangeDetails().add(eamProductAssetModelDTO);
                if (eamMtUseApplyDetail.getDetailType() == 3)
                    dto.getApplyDetails().add(eamProductAssetModelDTO);
            }
        }

        return dto;
    }


    /**
     * 根据主键获取量检具领用申请信息
     *
     * @param code 主键
     * @return StandardCodeDTO
     */
    public EamMtUseApplyDTO getBycode(String code) {
        // 查询数据
        QueryWrapper<EamMtUseApply> eamMtUseApplyQueryWrapper = new QueryWrapper<>();
        eamMtUseApplyQueryWrapper.lambda().eq(EamMtUseApply::getRecordCode, code);
        EamMtUseApply item = this.baseMapper.selectOne(eamMtUseApplyQueryWrapper);

        EamMtUseApplyDTO dto = new EamMtUseApplyDTO();
        try {
            BeanUtils.copyProperties(item, dto);
        } catch (RuntimeException e) {
            throw e;
        }

        QueryWrapper<EamMtUseApplyDetail> queryWrapper = new QueryWrapper<>();
        queryWrapper.lambda().eq(EamMtUseApplyDetail::getApplyId, dto.getId());
        List<EamMtUseApplyDetail> list = eamMtUseApplyDetailService.list(queryWrapper);

        List<String> ids = list.stream().map(EamMtUseApplyDetail::getAssetTypeId).collect(Collectors.toList());

        Map<String, EamAssetType> map = new HashMap<>();
        if (ids.size() > 0) {
            QueryWrapper<EamAssetType> eamAssetTypeQueryWrapper = new QueryWrapper<>();
            eamAssetTypeQueryWrapper.lambda().in(EamAssetType::getId, ids);
            List<EamAssetType> eamAssetTypes = eamAssetTypeService.list(eamAssetTypeQueryWrapper);
            for (EamAssetType eamAssetType : eamAssetTypes) {
                map.put(eamAssetType.getId(), eamAssetType);
            }
        }

        for (EamMtUseApplyDetail eamMtUseApplyDetail : list) {
            EamProductAssetModelDTO eamProductAssetModelDTO = new EamProductAssetModelDTO();
            eamProductAssetModelDTO.setApplyAmount(eamMtUseApplyDetail.getApplyAmount());
            eamProductAssetModelDTO.setAssetTypeId(eamMtUseApplyDetail.getAssetTypeId());
            eamProductAssetModelDTO.setAssetTypeName(map.get(eamMtUseApplyDetail.getAssetTypeId()).getAssetTypeName());
            eamProductAssetModelDTO.setAssetTypeCode(map.get(eamMtUseApplyDetail.getAssetTypeId()).getAssetTypeCode());
            if (dto.getApplyType() == 1) {
                dto.getApplyDetails().add(eamProductAssetModelDTO);
            } else {
                if (eamMtUseApplyDetail.getDetailType() == 1)
                    dto.getCurrentDetails().add(eamProductAssetModelDTO);
                if (eamMtUseApplyDetail.getDetailType() == 2)
                    dto.getChangeDetails().add(eamProductAssetModelDTO);
                if (eamMtUseApplyDetail.getDetailType() == 3)
                    dto.getApplyDetails().add(eamProductAssetModelDTO);
            }
        }

        return dto;
    }


    /**
     * 根据主键获取量检具领用申请信息
     *
     * @param id 主键
     * @return StandardCodeDTO
     */
    public void deleteitem(String id) {
        this.removeById(id);
    }
}

